package ch02.demo;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import java.io.FileOutputStream;

public class Demo05 {
    public static void main(String[] args) {
        try {
            //得到DOM解析器的工厂实例
            DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
            //从DOM工厂获得DOM解析器
            DocumentBuilder db = dbf.newDocumentBuilder();
            //解析XML文档，得到一个Document对象，即DOM树
            Document doc = db.parse("src/ch02/收藏信息2.xml");
            //找到要修改的节点
            NodeList list = doc.getElementsByTagName("Brand");
            for (int i = 0; i < list.getLength(); i++) {
                Element brandElement = (Element) list.item(i);
                String brandName = brandElement.getAttribute("name");
                if (brandName.equals("三星")) {
                    brandElement.setAttribute("name", "SAMSUNG"); //修改属性
                }
            }
            //保存xml文件
            TransformerFactory transformerFactory = TransformerFactory.newInstance();
            Transformer transformer = transformerFactory.newTransformer();
            DOMSource domSource = new DOMSource(doc);
            //设置编码类型
            transformer.setOutputProperty(OutputKeys.ENCODING, "utf-8");
            StreamResult result = new StreamResult(new FileOutputStream("src/ch02/收藏信息2.xml"));
            //把DOM树转换为XML文件
            transformer.transform(domSource, result);
        }catch (Exception e){
            e.printStackTrace();
        }
    }
}
